vae from scratch